package com.github.shiqiyue.app.config.webmvc;

import com.github.shiqiyue.app.common.utils.JsonUtils;
import io.micrometer.core.instrument.Metrics;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.method.HandlerMethod;
import org.springframework.web.multipart.MultipartHttpServletRequest;
import org.springframework.web.multipart.MultipartRequest;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.TimeUnit;

/***
 * 日志拦截器，进入方法前和方法执行完成后，写入日志；
 *
 * @author wenyao.wu
 *
 *
 */
public class LoggerInterceptor implements HandlerInterceptor {

    private ThreadLocal<Long> startTime = new ThreadLocal<>();
    private static final Logger logger = LoggerFactory.getLogger("requestMetric");
    private boolean enable = true;

    /***
     * 忽略的链接地址
     */
    private List<String> ignorePathList = new ArrayList<>();

    /**
     * preHandle方法是进行处理器拦截用的，顾名思义，该方法将在Controller处理之前进行调用，
     * SpringMVC中的Interceptor拦截器是链式的，可以同时存在
     * 多个Interceptor，然后SpringMVC会根据声明的前后顺序一个接一个的执行
     * ，而且所有的Interceptor中的preHandle方法都会在
     * Controller方法调用之前调用。SpringMVC的这种Interceptor链式结构也是可以进行中断的
     * ，这种中断方式是令preHandle的返 回值为false，当preHandle的返回值为false的时候整个请求就结束了。
     */
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler)
            throws Exception {
        if (!isProcess(request, response, handler)) {
            return true;
        }
        startTime.set(System.currentTimeMillis());
        if (logger.isInfoEnabled()) {
            try{
                HandlerMethod method = (HandlerMethod) handler;
                long threadId = Thread.currentThread().getId();
                StringBuffer sb = new StringBuffer();
                sb.append("url:").append(request.getServletPath()).append("||");
                sb.append("class:").append(method.getBean().getClass().getSimpleName()).append("||method:")
                        .append(method.getMethod().getName()).append("||thread id:").append(threadId)
                        .append("||start----");
                logger.info(sb.toString());
                logger.info("请求参数:{}", JsonUtils.beanToString(request.getParameterMap()));
            }catch (Exception e){

            }

        }
        return true;
    }

    /**
     * 这个方法只会在当前这个Interceptor的preHandle方法返回值为true的时候才会执行。postHandle是进行处理器拦截用的，
     * 它的执行时间是在处理器进行处理之
     * 后，也就是在Controller的方法调用之后执行，但是它会在DispatcherServlet进行视图的渲染之前执行
     * ，也就是说在这个方法中你可以对ModelAndView进行操
     * 作。这个方法的链式结构跟正常访问的方向是相反的，也就是说先声明的Interceptor拦截器该方法反而会后调用
     * ，这跟Struts2里面的拦截器的执行过程有点像，
     * 只是Struts2里面的intercept方法中要手动的调用ActionInvocation的invoke方法
     * ，Struts2中调用ActionInvocation的invoke方法就是调用下一个Interceptor
     * 或者是调用action，然后要在Interceptor之前调用的内容都写在调用invoke之前
     * ，要在Interceptor之后调用的内容都写在调用invoke方法之后。
     */
    @Override
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler,
                           ModelAndView modelAndView) throws Exception {

    }

    /***
     * 是否需要处理
     * @param request
     * @param response
     * @param handler
     * @return
     */
    private boolean isProcess(HttpServletRequest request, HttpServletResponse response, Object handler) {
        if (handler instanceof HandlerMethod) {
            if (request instanceof MultipartHttpServletRequest) {
                return false;
            }
            if (request instanceof MultipartRequest) {
                return false;
            }
            if (isEnable()) {
                return true;
            }
        }
        return false;
    }

    /**
     * 该方法也是需要当前对应的Interceptor的preHandle方法的返回值为true时才会执行。该方法将在整个请求完成之后，
     * 也就是DispatcherServlet渲染了视图执行，
     * 这个方法的主要作用是用于清理资源的，当然这个方法也只能在当前这个Interceptor的preHandle方法的返回值为true时才会执行。
     */
    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex)
            throws Exception {
        if (!isProcess(request, response, handler)) {
            return;
        }
        if (logger.isInfoEnabled()) {
            try{
                HandlerMethod method = (HandlerMethod) handler;
                long threadId = Thread.currentThread().getId();
                StringBuffer sb = new StringBuffer();
                sb.append("class:").append(method.getBean().getClass().getSimpleName()).append("||method:")
                        .append(method.getMethod().getName()).append("||thread id:").append(threadId)
                        .append("||use time:").append(System.currentTimeMillis() - startTime.get()).append("ms");
                logger.info(sb.toString());
            }catch (Exception e){

            }

        }
        startTime.remove();
    }

    public boolean isEnable() {
        return enable;
    }

    public void setEnable(boolean enable) {
        this.enable = enable;
    }

}
